import torch

# requires_grad 表示是否可以计算梯度
x = torch.ones((2, 2), requires_grad=True)

y = x + 2
z = y * y * 3
out = z.mean()

# .backward() 计算梯度的值
out.backward()

print(x.grad)
